package com.macro.mall;

import java.util.Properties;
import org.mybatis.generator.api.IntrospectedColumn;
import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.dom.java.CompilationUnit;
import org.mybatis.generator.api.dom.java.Field;
import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType;
import org.mybatis.generator.internal.DefaultCommentGenerator;
import org.mybatis.generator.internal.util.StringUtility;

/** 自定义注释生成器 Created by macro on 2018/4/26. */
public class CommentGenerator extends DefaultCommentGenerator {
  private static final String EXAMPLE_SUFFIX = "Example";

  private static final String MAPPER_SUFFIX = "Mapper";

  private static final String API_MODEL_PROPERTY_FULL_CLASS_NAME =
      "io.swagger.annotations.ApiModelProperty";

  private boolean addRemarkComments = false;

  /** 设置用户配置的参数 */
  @Override
  public void addConfigurationProperties(Properties properties) {
    super.addConfigurationProperties(properties);
    this.addRemarkComments = StringUtility.isTrue(properties.getProperty("addRemarkComments"));
  }

  /** 给字段添加注释 */
  @Override
  public void addFieldComment(
      Field field, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) {
    String remarks = introspectedColumn.getRemarks();
    // 根据参数和备注信息判断是否添加swagger注解信息
    if (addRemarkComments && StringUtility.stringHasValue(remarks)) {
      //            addFieldJavaDoc(field, remarks);
      // 数据库中特殊字符需要转义
      if (remarks.contains("\"")) {
        remarks = remarks.replace("\"", "'");
      }
      // 给model的字段添加swagger注解
      field.addJavaDocLine("@ApiModelProperty(value = \"" + remarks + "\")");
    }
  }

  /** 给model的字段添加注释 */
  private void addFieldJavaDoc(Field field, String remarks) {
    // 文档注释开始
    field.addJavaDocLine("/**");
    // 获取数据库字段的备注信息
    String[] remarkLines = remarks.split(System.getProperty("line.separator"));
    for (String remarkLine : remarkLines) {
      field.addJavaDocLine(" * " + remarkLine);
    }
    addJavadocTag(field, false);
    field.addJavaDocLine(" */");
  }

  @Override
  public void addJavaFileComment(CompilationUnit compilationUnit) {
    super.addJavaFileComment(compilationUnit);
    // 只在model中添加swagger注解类的导入
    if (!compilationUnit.getType().getFullyQualifiedName().contains(MAPPER_SUFFIX)
        && !compilationUnit.getType().getFullyQualifiedName().contains(EXAMPLE_SUFFIX)) {
      compilationUnit.addImportedType(
          new FullyQualifiedJavaType(API_MODEL_PROPERTY_FULL_CLASS_NAME));
    }
  }
}
